package com.juanjuan.myproject.utils;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import org.junit.Test;

/**
 * 代码生成工具类
 */
public class CodeGenerator {

    @Test
    public void run() {
        FastAutoGenerator.create("jdbc:mysql://124.0.0.1:3306/reggie?serverTimezone=Asia/Shanghai&useUnicode=true&characterEncoding=utf8&useSSL=false", "root", "zhao")
                //全局配置
                .globalConfig(builder -> {
                    builder.author("zhaocun") // 设置作者
                            .enableSwagger() // 开启 swagger 模式
                            .fileOverride() // 覆盖已生成文件
                            .outputDir("F:\\myproject\\src\\main\\java"); // 指定输出目录
                })
                //包配置
                .packageConfig(builder -> {
                    builder.parent("com.juanjuan") // 设置父包名
                            .moduleName("myproject")
                    .service(""); // 设置父包模块名
                })
                //策略配置
                .strategyConfig(builder -> {
                    builder.entityBuilder().idType(IdType.AUTO);
                    builder.serviceBuilder().formatServiceFileName("%sService");
                    builder.addInclude("rule") // 设置需要生成的表名
                            .addTablePrefix("t_", "c_"); // 设置过滤表前缀
                })
                //.templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板，默认的是Velocity引擎模板
                .execute();

    }



}
